import requests
from unit__001.unit_01.Tools2 import Tools2
class Tools():
    # url是接收的注册接口地址,username和password分别是用于注册的用户名和密码参数
    def register(self, url, username, password):
        # 在传入的 url 基础上拼接 /api/register
        url = f"{url}/api/register"
        # 创建一个字典 json 将传入的 username 和 password 作为键值对存入字典
        json = {'username': username, 'password': password}
        # url参数指定请求的URL地址，json参数指定请求体
        # 服务器返回的响应赋值给response
        response = Tools2().post(url=url,jsondata={'username':username,'password':password})
        return response

    def login(self, url, username, password):
        # 在传入的 url 基础上拼接 /api/login
        url = f"{url}/api/public/login"
        # 创建一个字典 json 将传入的 username 和 password 作为键值对存入字典
        json = {"username": username, "password": password}
        response = Tools2().post(url=url, jsondata={'username': username, 'password': password})
        token = None
        # 检查json中是否存在data,并且data对应的字典中是否存在token，如果都满足就提取token
        if "data" in response.json() and 'token' in response.json()['data']:
            token = response.json()['data']['token']
        return response, token



